package lee.t.code.str;

import org.junit.Assert;
import org.junit.Test;

/**
 * 外观数列
 * 给定一个正整数 n ，输出外观数列的第 n 项。
 * <p>
 * 「外观数列」是一个整数序列，从数字 1 开始，序列中的每一项都是对前一项的描述。
 * <p>
 * 你可以将其视作是由递归公式定义的数字字符串序列：
 * <p>
 * countAndSay(1) = "1"
 * countAndSay(n) 是对 countAndSay(n-1) 的描述，然后转换成另一个数字字符串。
 * 前五项如下：
 * <p>
 * 1.     1
 * 2.     11
 * 3.     21
 * 4.     1211
 * 5.     111221
 * 第一项是数字 1
 * 描述前一项，这个数是 1 即 “ 一 个 1 ”，记作 "11"
 * 描述前一项，这个数是 11 即 “ 二 个 1 ” ，记作 "21"
 * 描述前一项，这个数是 21 即 “ 一 个 2 + 一 个 1 ” ，记作 "1211"
 * 描述前一项，这个数是 1211 即 “ 一 个 1 + 一 个 2 + 二 个 1 ” ，记作 "111221"
 * 要 描述 一个数字字符串，首先要将字符串分割为 最小 数量的组，每个组都由连续的最多 相同字符 组成。然后对于每个组，先描述字符的数量，然后描述字符，形成一个描述组。要将描述转换为数字字符串，先将每组中的字符数量用数字替换，再将所有描述组连接起来。
 * <p>
 * 例如，数字字符串 "3322251" 的描述如下图：
 * 两个3, 叁个2, 一个 5, 一个 1 ===== 23 32, 15, 11
 * <p>
 * 提示：
 * <p>
 * 1 <= n <= 30
 * <p>
 * 作者：力扣 (LeetCode)
 * 链接：https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/xnpvdm/
 */
public class CountAndSay {
    enum Solution {
        loop {
            @Override
            public String countAndSay(int n) {
                StringBuilder sb = new StringBuilder("1");
                int offset = 0;
                for (int i = 1; i < n; i++) {
                    int from = offset;
                    offset = sb.length();
                    char c = sb.charAt(from);
                    int count = 0;
                    for (int j = from; j < offset; j++) {
                        char c1 = sb.charAt(j);
                        if (c == c1) {
                            count++;
                        } else {
                            sb.append(count).append(c);
                            c = c1;
                            count = 1;
                        }
                    }
                    sb.append(count).append(c);
                }
                return sb.substring(offset);
            }
        },
        recursion {
            @Override
            public String countAndSay(int n) {
                if (n <= 1) {
                    return "1";
                }
                String v = countAndSay(n - 1);
                int count = 0;
                char c = v.charAt(0);
                String res = "";
                for (int i = 0; i < v.length(); i++) {
                    if (v.charAt(i) == c) {
                        count++;
                    } else {
                        res += "" + count + c;
                        c = v.charAt(i);
                        count = 1;
                    }
                }
                res += "" + count + c;
                return res;
            }
        },
        recursionStringBuilder {
            @Override
            public String countAndSay(int n) {
                StringBuilder builder = new StringBuilder();
                int say = say(builder, n);
                return builder.substring(say);
            }

            public int say(StringBuilder str, int n) {
                if (n <= 1) {
                    str.append("1");
                    return 0;
                }
                int from = say(str, n - 1);
                int count = 0;
                char c = str.charAt(from);
                int len = str.length();
                for (int i = from; i < len; i++) {
                    if (str.charAt(i) == c) {
                        count++;
                    } else {
                        str.append(count).append(c);
                        c = str.charAt(i);
                        count = 1;
                    }
                }
                str.append(count).append(c);
                return len;
            }
        },
        ;

        public abstract String countAndSay(int n);
    }

    @Test
    public void test() {
        for (Solution value : Solution.values()) {
            String v = value.countAndSay(17);
            Assert.assertEquals(v,
                    "11131221131211132221232112111312212321123113112221121113122113111231133221121321132132211331121321231231121113122113322113111221131221");
        }
    }
}
